Enable AVX256 on AMD CPU (#1360)

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
microsoft · Nov 23, 2021 · a637cc2 · a637cc2
1 parent 1bc13fe
commit a637cc2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -302,6 +302,9 @@ def simd_width(self):
                 return '-D__AVX512__'
             elif 'avx2' in result:
                 return '-D__AVX256__'
+        elif 'authenticamd' in result:
+            if 'avx2' in result:
+                return '-D__AVX256__'
         return '-D__SCALAR__'
 
     def python_requirements(self):