Home
last modified time | relevance | path

Searched refs:getCurrentThreadIndices (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseSemiStructuredApplyDense.cu39 CUTLASS_DEVICE Tile8x8Masks* getCurrentThreadIndices() const {
60 Tile8x8Masks indices = *p.getCurrentThreadIndices();
DSparseSemiStructuredPack.h156 CUTLASS_DEVICE Tile8x8Masks* getCurrentThreadIndices() const { in getCurrentThreadIndices() function
405 *p.getCurrentThreadIndices() = indices; in sparse_semi_structured_tile_kernel()
452 Tile8x8Masks indices = *p.getCurrentThreadIndices(); in sparse_semi_structured_apply_kernel()