HomePhabricator

[CUDA] Implemented _[bi]mma* builtins.

Authored by tra on Apr 25 2019, 3:28 PM.

Description

[CUDA] Implemented _[bi]mma* builtins.

These builtins provide access to the new integer and
sub-integer variants of MMA (matrix multiply-accumulate) instructions
provided by CUDA-10.x on sm_75 (AKA Turing) GPUs.

Also added a feature for PTX 6.4. While Clang/LLVM does not generate
any PTX instructions that need it, we still need to pass it through to
ptxas in order to be able to compile code that uses the new 'mma'
instruction as inline assembly (e.g used by NVIDIA's CUTLASS library
https://github.com/NVIDIA/cutlass/blob/master/cutlass/arch/mma.h#L101)

Differential Revision: https://reviews.llvm.org/D60279

llvm-svn: 359248

Details

Committed
traApr 25 2019, 3:28 PM
Differential Revision
D60279: [CUDA] Implemented _[bi]mma* builtins.
Parents
rG16737538f4fc: PTX 6.3 extends `wmma` instruction to support s8/u8/s4/u4/b1 -> s32.
Branches
Unknown
Tags
Unknown