Updates for 3.4 release. (#1305)

2024-01-16 10:42:51 -08:00
parent acba5beee5
commit 2f589ffa76
166 changed files with 5996 additions and 4702 deletions
--- a/python/cutlass/emit/pytorch.py
+++ b/python/cutlass/emit/pytorch.py
@@ -657,7 +657,10 @@ class _ArchListSetter:
        """
        Restores the old value of TORCH_CUDA_ARCH_LIST
        """
-        os.environ[_ArchListSetter._TORCH_CUDA_ARCH_LIST] = self.old_arch_list
+        if self.old_arch_list is None:
+            del os.environ[_ArchListSetter._TORCH_CUDA_ARCH_LIST]
+        else:
+            os.environ[_ArchListSetter._TORCH_CUDA_ARCH_LIST] = self.old_arch_list


 def _jit(name: str, cc: int, cpp_file: str, cuda_file: str):