"sgl-kernel/vscode:/vscode.git/clone" did not exist on "23afdfd1c2b1c769eca97e2ecb2c595b1e18506a"
Unverified Commit 871f7e63 authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

Update setup.py (#298)

parent 1f972427
...@@ -14,7 +14,7 @@ from setuptools import setup, find_packages ...@@ -14,7 +14,7 @@ from setuptools import setup, find_packages
from torch.utils.cpp_extension import CUDAExtension, BuildExtension from torch.utils.cpp_extension import CUDAExtension, BuildExtension
cmdclass = {} cmdclass = {}
cmdclass['build_ext'] = BuildExtension cmdclass['build_ext'] = BuildExtension.with_options(use_ninja=False)
TORCH_MAJOR = int(torch.__version__.split('.')[0]) TORCH_MAJOR = int(torch.__version__.split('.')[0])
TORCH_MINOR = int(torch.__version__.split('.')[1]) TORCH_MINOR = int(torch.__version__.split('.')[1])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment