joviliast af15da2f84 Support WMMA layout in TritonAMDGPUAccelerateMatmulPass ...

-Introduce WmmaEncodingAttr for WMMA output
-Introduce BlockedToWMMA rewrite pattern in TritonAMDGPUAccelerateMatmulPass
-Provide a flag tho check if wmma instructions are supported by target

Signed-off-by: joviliast <iveselov.nn@gmail.com>

2023-12-18 09:11:20 -06:00

..

examples

Reformat Python code with yapf. (#2589)

2023-11-02 20:44:17 -07:00

perf-kernels

Add support for ALiBi-style attention bias (#417)

2023-12-15 16:28:37 -06:00

src

Support WMMA layout in TritonAMDGPUAccelerateMatmulPass

2023-12-18 09:11:20 -06:00

test

ROCM IFU: disable test_reduce_layouts

2023-12-13 17:12:39 -06:00

triton

Support WMMA layout in TritonAMDGPUAccelerateMatmulPass

2023-12-18 09:11:20 -06:00

tutorials

Initial commit to resolve merge conflicts

2023-11-20 22:41:03 +00:00

MANIFEST.in

[TOOLS] improved ahead-of-time compiler (#1805)

2023-06-21 01:02:58 -07:00

pyproject.toml

Enable ruff linter instead of flake8 (#2574)

2023-10-31 21:28:24 +00:00

setup.py

Initial commit to resolve merge conflicts

2023-11-20 22:41:03 +00:00