From ff5de59d23cfd9b08dac5cfa1fc9bd0fa09dab28 Mon Sep 17 00:00:00 2001 From: Jeff Rasley Date: Tue, 30 Aug 2022 09:36:33 -0700 Subject: [PATCH] prevent torch ext folder mkdir at tmp --- op_builder/builder.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/op_builder/builder.py b/op_builder/builder.py index 7067d8355c6a..4fc1d40eff53 100644 --- a/op_builder/builder.py +++ b/op_builder/builder.py @@ -487,13 +487,6 @@ def jit_load(self, verbose=True): self.jit_mode = True from torch.utils.cpp_extension import load - # Ensure directory exists to prevent race condition in some cases - ext_path = os.path.join( - os.environ.get('TORCH_EXTENSIONS_DIR', - DEFAULT_TORCH_EXTENSION_PATH), - self.name) - os.makedirs(ext_path, exist_ok=True) - start_build = time.time() sources = [self.deepspeed_src_path(path) for path in self.sources()] extra_include_paths = [