Skip to content

Commit 95723bc

Browse files
tdoublepGitHub Enterprise
authored andcommitted
Disable building C extensions when using target device sendnn (#34)
Upstream changes broke the logic to disable building the C extensions when building for sendnn. This was causing the following error when trying to pip install vllm inside the [dev container build](https://v3.travis.ibm.com/github/ai-foundation/aiu-inference-dev/builds/25651578). This tiny fix resolves that. Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>
1 parent bd8457a commit 95723bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ def _build_custom_ops() -> bool:
286286

287287

288288
def _build_core_ext() -> bool:
289-
return not (_is_neuron() or _is_tpu() or _is_openvino() or _is_xpu())
289+
return not (_is_neuron() or _is_tpu() or _is_openvino() or _is_xpu() or _is_sendnn())
290290

291291

292292
def get_hipcc_rocm_version():

0 commit comments

Comments
 (0)