Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/ci-gpu-rdna.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ jobs:
run: |
# Install User libraries
sudo apt-get update
sudo apt install -y libnuma1 numactl gfortran build-essential binutils dwarfdump
sudo apt install -y dwarfdump

# Install torch+rocm6.4
python -m pip install --upgrade pip
Expand Down
8 changes: 4 additions & 4 deletions .github/workflows/ci-gpu.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -59,8 +59,8 @@ jobs:
- name: Setup env
if: steps.cache-llvm-mlir.outputs.cache-hit != 'true'
run: |
sudo apt update
sudo apt install -y ninja-build cmake clang lld dwarfdump
sudo apt-get update
sudo apt-get install -y ninja-build cmake clang lld dwarfdump
pip install -r water/requirements-dev.txt

- name: Checkout LLVM
Expand Down Expand Up @@ -276,8 +276,8 @@ jobs:

- name: Setup env
run: |
sudo apt update
sudo apt install -y ninja-build cmake clang lld
sudo apt-get update
sudo apt-get install -y ninja-build cmake clang lld
pip install -r water/requirements-dev.txt

- name: Setup Cache Vars
Expand Down
2 changes: 1 addition & 1 deletion tests/kernel/wave_gemm_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ def get_test_shapes(test_name: str) -> list[tuple[int]]:
],
)
def testGemmBench(tmp_path, mfma_variant: MMAType, threads_per_wave: int):
shape = (64, 64, 64)
shape = (128, 128, 128)
perf_filename_tk = tmp_path / "wave_gemm_bench.txt"
perf_filename_iree = tmp_path / "iree_gemm_bench.txt"
enable_scheduling = SchedulingType.NONE
Expand Down
Loading