Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DNvGpuSupport.cpp26 constexpr int64_t kNumRowsPerTile = 8; variable
37 return (shape[0] / kNumRowsPerTile) * in inferNumRegistersPerMatrixFragment()
47 return {operandShape[0] / kNumRowsPerTile, in getTileShape()
224 params.numTiles = (shape[0] / kNumRowsPerTile) * in getLdMatrixParams()
227 params.numTiles = (shape[1] / kNumRowsPerTile) * in getLdMatrixParams()
241 const int64_t kNumThreadsPerTile = kNumRowsPerTile; in getLaneIdToLdMatrixMatrixCoord()
271 tileRow * kNumRowsPerTile + (d0 % kNumRowsPerTile)}); in getLaneIdToLdMatrixMatrixCoord()