#include "CGBuiltin.h"
#include "clang/Basic/TargetBuiltins.h"
#include "llvm/IR/IntrinsicsNVPTX.h"
Go to the source code of this file.
|
#define | MMA_INTR(geom_op_type, layout) Intrinsic::nvvm_wmma_##geom_op_type##_##layout##_stride |
|
#define | MMA_LDST(n, geom_op_type) { n, MMA_INTR(geom_op_type, col), MMA_INTR(geom_op_type, row) } |
|
#define | MMA_VARIANTS(geom, type) |
|
#define | MMA_SATF_VARIANTS(geom, type) |
|
#define | MMA_VARIANTS_I4(geom, type) |
|
#define | MMA_VARIANTS_B1_XOR(geom, type) |
|
#define | MMA_VARIANTS_B1_AND(geom, type) |
|
◆ MMA_INTR
#define MMA_INTR |
( |
|
geom_op_type, |
|
|
|
layout |
|
) |
| Intrinsic::nvvm_wmma_##geom_op_type##_##layout##_stride |
◆ MMA_LDST
#define MMA_LDST |
( |
|
n, |
|
|
|
geom_op_type |
|
) |
| { n, MMA_INTR(geom_op_type, col), MMA_INTR(geom_op_type, row) } |
◆ MMA_SATF_VARIANTS
#define MMA_SATF_VARIANTS |
( |
|
geom, |
|
|
|
type |
|
) |
| |
Value:
Intrinsic::nvvm_wmma_##geom##_mma_row_row_##
type##_satfinite, \
Intrinsic::nvvm_wmma_##geom##_mma_row_col_##
type##_satfinite, \
Intrinsic::nvvm_wmma_##geom##_mma_col_row_##
type##_satfinite, \
Intrinsic::nvvm_wmma_##geom##_mma_col_col_##
type##_satfinite
#define MMA_VARIANTS(geom, type)
const internal::VariadicAllOfMatcher< Type > type
Matches Types in the clang AST.
◆ MMA_VARIANTS
#define MMA_VARIANTS |
( |
|
geom, |
|
|
|
type |
|
) |
| |
Value: Intrinsic::nvvm_wmma_##geom##_mma_row_row_##
type, \
Intrinsic::nvvm_wmma_##geom##_mma_row_col_##
type, \
Intrinsic::nvvm_wmma_##geom##_mma_col_row_##
type, \
Intrinsic::nvvm_wmma_##geom##_mma_col_col_##
type
◆ MMA_VARIANTS_B1_AND
#define MMA_VARIANTS_B1_AND |
( |
|
geom, |
|
|
|
type |
|
) |
| |
Value: 0, \
Intrinsic::nvvm_wmma_##geom##_mma_and_popc_row_col_##
type, \
0, \
0, \
0, \
0, \
0, \
0
◆ MMA_VARIANTS_B1_XOR
#define MMA_VARIANTS_B1_XOR |
( |
|
geom, |
|
|
|
type |
|
) |
| |
Value: 0, \
Intrinsic::nvvm_wmma_##geom##_mma_xor_popc_row_col_##
type, \
0, \
0, \
0, \
0, \
0, \
0
◆ MMA_VARIANTS_I4
#define MMA_VARIANTS_I4 |
( |
|
geom, |
|
|
|
type |
|
) |
| |
Value: 0, \
Intrinsic::nvvm_wmma_##geom##_mma_row_col_##
type, \
0, \
0, \
0, \
Intrinsic::nvvm_wmma_##geom##_mma_row_col_##
type##_satfinite, \
0, \
0