File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments