Skip to content

Commit 4148828

Browse files
author
Peter Yeh
committed
Refactor: Rename this_dir to curdir in setup.py
Minor variable renaming for clarity in extension directory path resolution
1 parent 135c2b2 commit 4148828

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

setup.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -247,8 +247,8 @@ def get_extensions():
247247
extra_compile_args["nvcc"].append("-g")
248248
extra_link_args.append("/DEBUG")
249249

250-
this_dir = os.path.dirname(os.path.curdir)
251-
extensions_dir = os.path.join(this_dir, "torchao", "csrc")
250+
curdir = os.path.dirname(os.path.curdir)
251+
extensions_dir = os.path.join(curdir, "torchao", "csrc")
252252
sources = list(glob.glob(os.path.join(extensions_dir, "**/*.cpp"), recursive=True))
253253

254254
extensions_cuda_dir = os.path.join(extensions_dir, "cuda")
@@ -278,8 +278,8 @@ def get_extensions():
278278
]
279279
)
280280

281-
this_dir = os.path.dirname(os.path.curdir)
282-
extensions_dir = os.path.join(this_dir, "torchao", "csrc")
281+
curdir = os.path.dirname(os.path.curdir)
282+
extensions_dir = os.path.join(curdir, "torchao", "csrc")
283283
sources = list(glob.glob(os.path.join(extensions_dir, "**/*.cpp"), recursive=True))
284284

285285
extensions_cuda_dir = os.path.join(extensions_dir, "cuda")

0 commit comments

Comments
 (0)