Add environment variable to make nvcc compilation more verbose (#2759)

microsoft · Jan 28, 2023 · 258d283 · 258d283
1 parent cc3d7cb
commit 258d283
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -693,6 +693,8 @@ def nvcc_args(self):
                 '-U__CUDA_NO_HALF_CONVERSIONS__',
                 '-U__CUDA_NO_HALF2_OPERATORS__'
             ]
+            if os.environ.get('DS_DEBUG_CUDA_BUILD', '0') == '1':
+                args.append('--ptxas-options=-v')
             args += self.compute_capability_args()
         return args