Searched defs:InstructionShape (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseSemiStructuredLinear.cu | 719 using InstructionShape = cutlass::gemm::GemmShape<16, 8, 64>; in _sparse_semi_structured_linear() typedef
783 using InstructionShape = cutlass::gemm::GemmShape<16, 8, 32>; in _sparse_semi_structured_linear() typedef
822 using InstructionShape = cutlass::gemm::GemmShape<16, 8, 32>; in _sparse_semi_structured_linear() typedef
861 using InstructionShape = cutlass::gemm::GemmShape<16, 8, 16>; in _sparse_semi_structured_linear() typedef
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/warp/ |
| H A D | mma_tensorop_dequantizer.h | 111 using InstructionShape = typename ArchMmaOperator::Shape; variable
219 using InstructionShape = typename ArchMmaOperator::Shape; variable
314 using InstructionShape = typename ArchMmaOperator::Shape; variable
401 using InstructionShape = typename ArchMmaOperator::Shape; variable
|
| H A D | mma_tensorop_compute_B_with_f16.h | 141 using InstructionShape = typename ArchMmaOperator::Shape; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/ |
| H A D | mma_accum_lambda_iterator.h | 30 using InstructionShape = typename T::InstructionShape; member
103 using InstructionShape = typename T::InstructionShape; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/ |
| H A D | gemm_kernel_utils.h | 107 using InstructionShape = cutlass::gemm::GemmShape<1, 1, 1>; member
|
| H A D | kernel_backward.h | 364 using InstructionShape = typename GemmType::InstructionShape; member
499 using InstructionShape = typename GemmType::InstructionShape; member
551 using InstructionShape = typename GemmType::InstructionShape; member
|
| H A D | kernel_forward.h | 453 using InstructionShape = typename GemmType::InstructionShape; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/ |
| H A D | warp_iterator_from_smem.h | 85 using InstructionShape = InstructionShape_; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/kernel/ |
| H A D | fpA_intB_gemm.h | 84 using InstructionShape = typename Mma::Policy::Operator::InstructionShape; member
|