More flags

ROCm · Sep 12, 2024 · 7003b95 · 7003b95
1 parent 1c46c14
commit 7003b95
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/gradlib/setup.py b/gradlib/setup.py
@@ -130,7 +130,8 @@
                     'nvcc': [
                         '-O3', '-U__CUDA_NO_HALF_OPERATORS__',
                         '-U__CUDA_NO_HALF_CONVERSIONS__',
-                        "-ftemplate-depth=1024"
+                        "-ftemplate-depth=1024",
+                        '-DLEGACY_HIPBLAS_DIRECT=ON',
                     ] + extra_args
                 }))
         ext_modules.append(
@@ -148,7 +149,8 @@
                     'nvcc': [
                         '-O3', '-U__CUDA_NO_HALF_OPERATORS__',
                         '-U__CUDA_NO_HALF_CONVERSIONS__',
-                        "-ftemplate-depth=1024"
+                        "-ftemplate-depth=1024",
+                        '-DLEGACY_HIPBLAS_DIRECT=ON',
                     ] + extra_args
                 }))