From 5369df8331b908f4b6d64aec310dd094826fe3c1 Mon Sep 17 00:00:00 2001
From: Connor Holmes <connorholmes@microsoft.com>
Date: Sat, 28 Jan 2023 07:17:40 +0000
Subject: [PATCH] Add environment variable to make nvcc compilation more
 verbose

---
 op_builder/builder.py | 2 ++
 1 file changed, 2 insertions(+)

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