diff --git a/.github/workflows/validate-binaries.yml b/.github/workflows/validate-binaries.yml index 087f3130f4..c08e673064 100644 --- a/.github/workflows/validate-binaries.yml +++ b/.github/workflows/validate-binaries.yml @@ -43,5 +43,7 @@ jobs: os: "linux" channel: ${{ inputs.channel }} repository: "pytorch/ao" + with_cuda: "enable" + with-rocm: "disable" smoke_test: "source ./.github/scripts/validate_binaries.sh" install_torch: true