1 files changed, 5 insertions, 5 deletions

diff --git a/src/cube.h b/src/cube.h
index 363cad4..4c6de2f 100644
--- a/src/cube.h
+++ b/src/cube.h
@@ -1,11 +1,11 @@
 /* Types *********************************************************************/
 
-/* TODO: ifdef for different implementations */
 /* See doc/CUBE_INTERNAL.md for a description of the cube format */
-typedef struct {
-        uint8_t c[8];
-        uint8_t e[12];
-} cube_t;
+#ifdef CUBE_AVX2
+typedef __m256i cube_t;
+#else
+typedef struct { uint8_t c[8]; uint8_t e[12]; } cube_t;
+#endif
 
 typedef uint8_t move_t;
 typedef uint8_t trans_t;