Update setup.py (#298)

871f7e63 · Jeff Rasley · GitHub · 1f972427 · 871f7e63
Unverified Commit 871f7e63 authored Jul 22, 2020 by Jeff Rasley Committed by GitHub Jul 22, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@ from setuptools import setup, find_packages
 from torch.utils.cpp_extension import CUDAExtension, BuildExtension

 cmdclass = {}
-cmdclass['build_ext'] = BuildExtension
+cmdclass['build_ext'] = BuildExtension.with_options(use_ninja=False)

 TORCH_MAJOR = int(torch.__version__.split('.')[0])
 TORCH_MINOR = int(torch.__version__.split('.')[1])