We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 66d2a58 commit d96d838Copy full SHA for d96d838
setup.py
@@ -81,7 +81,8 @@ def build_cmake(self, ext: CMakeExtension):
81
if hasattr(torch, "cuda") and torch.cuda.is_available():
82
cmake_args += ["-DRUNTIME_ENVIRONMENT=cuda"]
83
else:
84
- cmake_args += ["-DRUNTIME_ENVIRONMENT=ascend"]
+ cmake_args += ["-DRUNTIME_ENVIRONMENT=simu"]
85
+ cmake_args += ["-DBUILD_UCM_SPARSE=OFF"]
86
87
subprocess.check_call(
88
["cmake", *cmake_args, ext.cmake_file_path], cwd=build_dir
0 commit comments