|
tvm
|
Static options for the block-scaled tcgen05 instruction descriptor. More...
#include <tcgen05.h>
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("tirx.cuda.TCGen05InstrDescriptorBlockScaledAttrs", TCGen05InstrDescriptorBlockScaledAttrs, AttrsNode) | |
Public Member Functions inherited from tvm::AttrsNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("ir.Attrs", AttrsNode, ffi::Object) | |
Static Public Member Functions | |
| static void | RegisterReflection () |
Public Attributes | |
| ffi::String | d_dtype |
| ffi::String | a_dtype |
| ffi::String | b_dtype |
| ffi::String | sfa_dtype |
| ffi::String | sfb_dtype |
| int64_t | M |
| int64_t | N |
| int64_t | K |
| bool | trans_a |
| bool | trans_b |
| int64_t | n_cta_groups = 1 |
| bool | neg_a = false |
| bool | neg_b = false |
| bool | is_sparse = false |
Additional Inherited Members | |
Static Public Attributes inherited from tvm::AttrsNode | |
| static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
Static options for the block-scaled tcgen05 instruction descriptor.
|
static |
| tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "tirx.cuda.TCGen05InstrDescriptorBlockScaledAttrs" | , |
| TCGen05InstrDescriptorBlockScaledAttrs | , | ||
| AttrsNode | |||
| ) |
| ffi::String tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::a_dtype |
| ffi::String tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::b_dtype |
| ffi::String tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::d_dtype |
| bool tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::is_sparse = false |
| int64_t tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::K |
| int64_t tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::M |
| int64_t tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::N |
| int64_t tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::n_cta_groups = 1 |
| bool tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::neg_a = false |
| bool tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::neg_b = false |
| ffi::String tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::sfa_dtype |
| ffi::String tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::sfb_dtype |
| bool tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::trans_a |
| bool tvm::backend::cuda::TCGen05InstrDescriptorBlockScaledAttrs::trans_b |