[Compilation] Add _NO_HALF2 flags to be consistent with Pytorch

eb7b89771e/cmake/Dependencies.cmake (L1693)
2023-01-12 22:15:41 -08:00 · 2023-01-12 22:15:41 -08:00 · d509832426
commit d509832426
parent f95c2fc108
1 changed files with 2 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@ -136,6 +136,8 @@ ext_modules.append(
                    "-std=c++17",
                    "-U__CUDA_NO_HALF_OPERATORS__",
                    "-U__CUDA_NO_HALF_CONVERSIONS__",
+                    "-U__CUDA_NO_HALF2_OPERATORS__",
+                    "-U__CUDA_NO_BFLOAT16_CONVERSIONS__",
                    "--expt-relaxed-constexpr",
                    "--expt-extended-lambda",
                    "--use_fast_math",