diff --git a/.github/workflows/build_and_test_maxtext.yml b/.github/workflows/build_and_test_maxtext.yml index d044e8493e..fe4e2a2f10 100644 --- a/.github/workflows/build_and_test_maxtext.yml +++ b/.github/workflows/build_and_test_maxtext.yml @@ -286,7 +286,7 @@ jobs: xla_python_client_mem_fraction: 0.9 tf_force_gpu_allow_growth: true container_resource_option: "--device=/dev/kfd --device=/dev/dri --group-add video --ipc=host --shm-size 64g --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --privileged" - requirements_file: "dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt" + requirements_file: "src/dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt" decoupled_mode: true is_scheduled_run: ${{ github.event_name == 'schedule' }} maxtext_sha: ${{ needs.build_and_upload_maxtext_package.outputs.maxtext_sha }} diff --git a/.github/workflows/build_rocm_transformer_engine_wheel_weekly.yml b/.github/workflows/build_rocm_transformer_engine_wheel_weekly.yml index aede854e9d..c6fdd677e7 100644 --- a/.github/workflows/build_rocm_transformer_engine_wheel_weekly.yml +++ b/.github/workflows/build_rocm_transformer_engine_wheel_weekly.yml @@ -46,7 +46,7 @@ jobs: run: | set -euo pipefail source .venv/bin/activate - uv pip install -r dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt + uv pip install -r src/dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt - name: Detect ROCm version and Python tag shell: bash diff --git a/dependencies/requirements/requirements_decoupled_jax_0_8_2.txt b/src/dependencies/requirements/requirements_decoupled_jax_0_8_2.txt similarity index 100% rename from dependencies/requirements/requirements_decoupled_jax_0_8_2.txt rename to src/dependencies/requirements/requirements_decoupled_jax_0_8_2.txt diff --git a/dependencies/requirements/requirements_decoupled_rocm_jax_0_7.1.txt b/src/dependencies/requirements/requirements_decoupled_rocm_jax_0_7.1.txt similarity index 100% rename from dependencies/requirements/requirements_decoupled_rocm_jax_0_7.1.txt rename to src/dependencies/requirements/requirements_decoupled_rocm_jax_0_7.1.txt diff --git a/dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt b/src/dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt similarity index 100% rename from dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt rename to src/dependencies/requirements/requirements_decoupled_rocm_jax_0_8_2.txt diff --git a/dependencies/requirements/requirements_rocm_jax_0.8.2.txt b/src/dependencies/requirements/requirements_rocm_jax_0.8.2.txt similarity index 100% rename from dependencies/requirements/requirements_rocm_jax_0.8.2.txt rename to src/dependencies/requirements/requirements_rocm_jax_0.8.2.txt