[BACKEND] add dot conversion (mma version=2) (#672)

Browse Source

LLVM Conversion for Dot op.

Due to the lack of `convert_layout`, currently, the dot only supports
the following combination of operands

- `$a` in shared layout
- `$b` in shared layout
- `$c` in MMA layout(but only Splat-like, leaving the generic cases to
`convert_layout`)

This PR focus on `mma.16816` related logic support, leaving the other
cases to the following PR.

Co-authored-by: Philippe Tillet <phil@openai.com>

...

This commit is contained in:

Yan Chunwei

2022-09-23 11:43:54 +08:00

committed by GitHub

parent 23f424c660

commit 922155f1d2

7 changed files with 1033 additions and 116 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

1075

lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp

File diff suppressed because it is too large Load Diff