Skip to content

Commit

Permalink
[Tcp] Update docker script to enable TCP and its e2e tests
Browse files Browse the repository at this point in the history
  • Loading branch information
navahgar committed Dec 12, 2022
1 parent 21f233b commit 925e534
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion build_tools/python_deploy/build_linux_packages.sh
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,7 @@ function build_in_tree() {
-DLLVM_TARGETS_TO_BUILD=host \
-DMLIR_ENABLE_BINDINGS_PYTHON=ON \
-DTORCH_MLIR_ENABLE_LTC=ON \
-DTORCH_MLIR_ENABLE_TCP=OFF \
-DTORCH_MLIR_ENABLE_TCP=ON \
-DTORCH_MLIR_USE_INSTALLED_PYTORCH="$torch_from_bin" \
-DTORCH_MLIR_SRC_PYTORCH_REPO=${TORCH_MLIR_SRC_PYTORCH_REPO} \
-DTORCH_MLIR_SRC_PYTORCH_BRANCH=${TORCH_MLIR_SRC_PYTORCH_BRANCH} \
Expand Down Expand Up @@ -277,6 +277,9 @@ function test_in_tree() {

echo ":::: Run TorchDynamo e2e integration tests"
python -m e2e_testing.main --config=torchdynamo -v

echo ":::: Run TCP e2e integration tests"
python -m e2e_testing.main --config=tcp -v
}

function setup_venv() {
Expand Down

0 comments on commit 925e534

Please sign in to comment.