ContextQMD
Back to Cutlass

CUTLASS: mma_tensor_op_tile_iterator.h File Reference

docs/mma__tensor__op__tile__iterator_8h.html

4.4.28.2 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Classes | Namespaces

mma_tensor_op_tile_iterator.h File Reference

Defines iterators used by warp-level matrix multiply operations targeting Tensor Cores. More...

#include "cutlass/cutlass.h"
#include "cutlass/array.h"
#include "cutlass/numeric_types.h"
#include "cutlass/tensor_ref.h"
#include "cutlass/matrix_shape.h"
#include "cutlass/arch/memory_sm75.h"
#include "cutlass/gemm/gemm.h"
#include "cutlass/layout/matrix.h"
#include "cutlass/layout/tensor.h"
#include "cutlass/layout/pitch_linear.h"
#include "cutlass/layout/tensor_op_multiplicand_sm75.h"
#include "cutlass/platform/platform.h"
#include "cutlass/fast_math.h"

Include dependency graph for mma_tensor_op_tile_iterator.h:

This graph shows which files directly or indirectly include this file:

[Go to the source code of this file.](mma tensor op tile iterator_8h_source.html)

|

Classes

| | class | cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand, Element_, Layout_, InstructionShape_, OpDelta_, Threads, PartitionsK_ > | | | | class | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::TensorOpMultiplicandCongruous< sizeof_bits< Element_ >::value, 64 >, InstructionShape_, OpDelta_, 32, PartitionsK_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 0ed7daaeba1c095e77f68533d4d2c475c.html) | | | | struct | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::TensorOpMultiplicandCongruous< sizeof_bits< Element_ >::value, 64 >, InstructionShape_, OpDelta_, 32, PartitionsK_ >::Policy](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 07638f8b7761f6e2e2e6918e2c05e739.html) | | | Internal structure of iterator - made public to enable introspection. [More...](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 07638f8b7761f6e2e2e6918e2c05e739.html#details)
| | | | class | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::ColumnMajorTensorOpMultiplicandCongruous< sizeof_bits< Element_ >::value, int(128/sizeof(Element_))>, InstructionShape_, OpDelta_, 32, PartitionsK_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 0b84f53cd44b339eccc12067c9f86e11c.html) | | | | class | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::RowMajorTensorOpMultiplicandCongruous< sizeof_bits< Element_ >::value, int(128/sizeof(Element_))>, InstructionShape_, OpDelta_, 32, PartitionsK_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 039819fb3ccd43786d556c2c9669508ef.html) | | | | class | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::TensorOpMultiplicandCrosswise< sizeof_bits< Element_ >::value, Crosswise >, InstructionShape_, OpDelta_, 32, PartitionsK_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 0c7d419c589d601ce4eb603be566fea21.html) | | | | struct | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::TensorOpMultiplicandCrosswise< sizeof_bits< Element_ >::value, Crosswise >, InstructionShape_, OpDelta_, 32, PartitionsK_ >::Policy](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 0784c74bd670999ec23ad8ef9dc55777.html) | | | Internal structure of iterator - made public to enable introspection. [More...](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 0784c74bd670999ec23ad8ef9dc55777.html#details)
| | | | class | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::ColumnMajorTensorOpMultiplicandCrosswise< sizeof_bits< Element_ >::value, Crosswise >, InstructionShape_, OpDelta_, 32, PartitionsK_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 0e52ad425e1ee3e68544873f66733237b.html) | | | | class | [cutlass::gemm::warp::MmaTensorOpMultiplicandTileIterator< Shape_, Operand_, Element_, cutlass::layout::RowMajorTensorOpMultiplicandCrosswise< sizeof_bits< Element_ >::value, Crosswise >, InstructionShape_, OpDelta_, 32, PartitionsK_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpMultiplicandTileIterator_3_01Shape 00_01Operand 0352e0dcab42bc8360606874e00173556.html) | | | | class | cutlass::gemm::warp::MmaTensorOpAccumulatorTileIterator< Shape_, Element_, Layout_, InstructionShape_, OpDelta_ > | | | | class | [cutlass::gemm::warp::MmaTensorOpAccumulatorTileIterator< Shape_, Element_, cutlass::layout::RowMajor, InstructionShape_, OpDelta_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 006c39f57875e0aa9d0ad82c8043ed8b98.html) | | | | struct | [cutlass::gemm::warp::MmaTensorOpAccumulatorTileIterator< Shape_, Element_, cutlass::layout::RowMajor, InstructionShape_, OpDelta_ >::Policy](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 093b5d2838ac5a742704ef62b5c8688f0.html) | | | Internal structure of iterator - made public to enable introspection. [More...](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 093b5d2838ac5a742704ef62b5c8688f0.html#details)
| | | | class | [cutlass::gemm::warp::MmaTensorOpAccumulatorTileIterator< Shape_, Element_, cutlass::layout::ColumnMajor, InstructionShape_, OpDelta_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 008f607b871a2b3d854eb4def64712c042.html) | | | | struct | [cutlass::gemm::warp::MmaTensorOpAccumulatorTileIterator< Shape_, Element_, cutlass::layout::ColumnMajor, InstructionShape_, OpDelta_ >::Policy](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 0d35fa5dc4e4b4f72784c943fd857fc1d.html) | | | Internal structure of iterator - made public to enable introspection. [More...](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 0d35fa5dc4e4b4f72784c943fd857fc1d.html#details)
| | | | class | [cutlass::gemm::warp::MmaTensorOpAccumulatorTileIterator< Shape_, Element_, cutlass::layout::ColumnMajorInterleaved< InterleavedN >, InstructionShape_, OpDelta_ >](classcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 00027dabdc144edd6276f664ca74088510.html) | | | | struct | [cutlass::gemm::warp::MmaTensorOpAccumulatorTileIterator< Shape_, Element_, cutlass::layout::ColumnMajorInterleaved< InterleavedN >, InstructionShape_, OpDelta_ >::Policy](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 03822d9be37f3725022005a5434441f22.html) | | | Internal structure of iterator - made public to enable introspection. [More...](structcutlass_1_1gemm_1_1warp_1_1MmaTensorOpAccumulatorTileIterator_3_01Shape 00_01Element 03822d9be37f3725022005a5434441f22.html#details)
| | |

|

Namespaces

| | | cutlass | | | | | cutlass::gemm | | | | | cutlass::gemm::warp | | |

Generated by 1.8.11