diff --git a/.github/workflows/amd_ci.yml b/.github/workflows/amd_ci.yml index 5816f0729..2ef300b66 100644 --- a/.github/workflows/amd_ci.yml +++ b/.github/workflows/amd_ci.yml @@ -117,4 +117,4 @@ jobs: source "${{ runner.tool_cache }}/${{ env.VENV_DIR }}/bin/activate" cd testing/python/amd unset PYTHONPATH - python -m pytest -v test_tilelang_test_amd.py + python -m pytest -v test_tilelang_test_amd.py \ No newline at end of file diff --git a/requirements-rocm.txt b/requirements-rocm.txt index 4c8df9c67..bdf1aa985 100644 --- a/requirements-rocm.txt +++ b/requirements-rocm.txt @@ -20,7 +20,6 @@ requests cloudpickle ml_dtypes psutil -torch tabulate wheel setuptools