Home
last modified time | relevance | path

Searched refs:AlignmentInputA (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseSemiStructuredLinear.cu68 constexpr int AlignmentInputA = 128 / cutlass::sizeof_bits<ElementInputA>::value;
162 AlignmentInputA,
DSparseSemiStructuredOps.cu63 constexpr int AlignmentInputA = 128 / cutlass::sizeof_bits<ElementInputA>::value;
154 AlignmentInputA,
/external/pytorch/aten/src/ATen/native/cuda/
DRowwiseScaledMM.cu159 constexpr int AlignmentInputA = 16 / sizeof(DtypeA); in f8f8bf16_rowwise_impl() local
227 AlignmentInputA, in f8f8bf16_rowwise_impl()