pytorch
diff --git a/‎test/test_cuda.py‎
Lines changed: 0 additions & 5 deletions b/‎test/test_cuda.py‎
Lines changed: 0 additions & 5 deletions
diff --git a/‎torch/cuda/__init__.py‎
Lines changed: 0 additions & 2 deletions b/‎torch/cuda/__init__.py‎
Lines changed: 0 additions & 2 deletions
@@ -6484,11 +6484,6 @@ def test_cuda_autocast_deprecated_warning(self):
             with torch.cuda.amp.autocast():
                 _ = torch.ones(10)
 
-    def test_cuda_module_loading_env(self):
-        torch.cuda.init()
-        val = os.environ.get("CUDA_MODULE_LOADING", "")
-        self.assertEqual(val, "LAZY")
-
 
 @unittest.skipIf(
     os.environ.get("USE_LEGACY_DRIVER", None) == "1", "Doesn't work with older driver"
 
@@ -379,8 +379,6 @@ def _lazy_init():
             )
         # This function throws if there's a driver initialization error, no GPUs
         # are found or any other error occurs
-        if "CUDA_MODULE_LOADING" not in os.environ:
-            os.environ["CUDA_MODULE_LOADING"] = "LAZY"
         torch._C._cuda_init()
         # Some of the queued calls may reentrantly call _lazy_init();
         # we need to just return without initializing in that case.
Original file line number	Diff line number	Diff line change
`@@ -379,8 +379,6 @@ def _lazy_init():`
`379`	`379`	`)`
`380`	`380`	`# This function throws if there's a driver initialization error, no GPUs`
`381`	`381`	`# are found or any other error occurs`
`382`		`- if "CUDA_MODULE_LOADING" not in os.environ:`
`383`		`- os.environ["CUDA_MODULE_LOADING"] = "LAZY"`
`384`	`382`	`torch._C._cuda_init()`
`385`	`383`	`# Some of the queued calls may reentrantly call _lazy_init();`
`386`	`384`	`# we need to just return without initializing in that case.`