Skip to content

Commit ca429dc

Browse files
authored
ci: check the ExecuTorch build job installed a wheel with the C++ runtime (#4554)
1 parent 0be4d77 commit ca429dc

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

.github/workflows/executorch-build-linux.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,13 @@ jobs:
8282
python -m pip install pyyaml "executorch==1.4.1"
8383
export TORCH_TENSORRT_EXECUTORCH_RUNTIME_VERSION="$(python -c 'import importlib.metadata; print(importlib.metadata.version("torch-tensorrt"))')"
8484
85+
# The downloaded wheel has to carry the C++ runtime. A wheel built with
86+
# PYTHON_ONLY=1 has no libtorchtrt.so, which leaves
87+
# ENABLED_FEATURES.torch_tensorrt_runtime False, and the export near the
88+
# end of this script then fails after the Bazel builds have already run.
89+
# Check it here so the wrong wheel is reported as the wrong wheel.
90+
python -c 'from torch_tensorrt._features import ENABLED_FEATURES as f; assert f.torch_tensorrt_runtime, f'
91+
8592
# Bazel truncates a failing action's output at 1 MB by default and then
8693
# prints nothing but the size, which hid the real error behind
8794
# stdout ... exceeds maximum size of

0 commit comments

Comments
 (0)