Skip to content

Commit 5d1a30c

Browse files
authored
DS_BUILD_OPS should build only compatible ops (#6489)
Currently DS_BUILD_OPS=1 fails on incompatible ops. This is a deviation from [documentation](https://www.deepspeed.ai/tutorials/advanced-install/#pre-install-deepspeed-ops) which states that only compatible ops are built. <img width="614" alt="image" src="https://github.com/user-attachments/assets/0f1a184e-b568-4d25-9e9b-e394fb047df2">
1 parent ddeb0c1 commit 5d1a30c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: setup.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -183,8 +183,8 @@ def op_enabled(op_name):
183183
if op_enabled(op_name) and not op_compatible:
184184
env_var = op_envvar(op_name)
185185
if not is_env_set(env_var):
186-
builder.warning(f"One can disable {op_name} with {env_var}=0")
187-
abort(f"Unable to pre-compile {op_name}")
186+
builder.warning(f"Skip pre-compile of incompatible {op_name}; One can disable {op_name} with {env_var}=0")
187+
continue
188188

189189
# If op is compatible but install is not enabled (JIT mode).
190190
if is_rocm_pytorch and op_compatible and not op_enabled(op_name):

0 commit comments

Comments
 (0)