Skip to content

Commit d96d838

Browse files
committed
fix build in cpu env
1 parent 66d2a58 commit d96d838

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,8 @@ def build_cmake(self, ext: CMakeExtension):
8181
if hasattr(torch, "cuda") and torch.cuda.is_available():
8282
cmake_args += ["-DRUNTIME_ENVIRONMENT=cuda"]
8383
else:
84-
cmake_args += ["-DRUNTIME_ENVIRONMENT=ascend"]
84+
cmake_args += ["-DRUNTIME_ENVIRONMENT=simu"]
85+
cmake_args += ["-DBUILD_UCM_SPARSE=OFF"]
8586

8687
subprocess.check_call(
8788
["cmake", *cmake_args, ext.cmake_file_path], cwd=build_dir

0 commit comments

Comments
 (0)