From a99d2e5df7d6df0d99d8d5bd4db99ecb9f1014e6 Mon Sep 17 00:00:00 2001 From: Tianyu Liu Date: Fri, 30 Aug 2024 13:15:41 -0700 Subject: [PATCH] remove float8 install as H100 is not available in CI yet ghstack-source-id: fbbf3fdd2d52334b76e45bac8b22a69de71baa5f Pull Request resolved: https://github.com/pytorch/torchtitan/pull/565 --- .github/workflows/integration_test_4gpu.yaml | 1 - 1 file changed, 1 deletion(-) diff --git a/.github/workflows/integration_test_4gpu.yaml b/.github/workflows/integration_test_4gpu.yaml index a82bd96c..993ae797 100644 --- a/.github/workflows/integration_test_4gpu.yaml +++ b/.github/workflows/integration_test_4gpu.yaml @@ -38,6 +38,5 @@ jobs: pip config --user set global.progress_bar off python -m pip install --force-reinstall --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 - USE_CPP=0 python -m pip install git+https://github.com/pytorch/ao.git mkdir artifacts-to-be-uploaded python ./test_runner.py artifacts-to-be-uploaded --ngpu 4