diff --git a/op_builder/builder.py b/op_builder/builder.py index 0ae428bdfcfc..198e8471a0b5 100644 --- 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