Lines Matching refs:params
210 LdMatrixParams params; in getLdMatrixParams() local
212 params.fragmentType = type.vectorType; in getLdMatrixParams()
215 params.targetLayout = NVVM::MMALayout::row; in getLdMatrixParams()
217 params.targetLayout = NVVM::MMALayout::col; in getLdMatrixParams()
220 params.contiguousDimType = in getLdMatrixParams()
223 if (params.contiguousDimType == IteratorType::Reduction) { in getLdMatrixParams()
224 params.numTiles = (shape[0] / kNumRowsPerTile) * in getLdMatrixParams()
227 params.numTiles = (shape[1] / kNumRowsPerTile) * in getLdMatrixParams()
231 if (params.numTiles == 0) in getLdMatrixParams()
234 return params; in getLdMatrixParams()
239 const LdMatrixParams ¶ms) { in getLaneIdToLdMatrixMatrixCoord() argument
243 params.fragmentType.getElementType().getIntOrFloatBitWidth()); in getLaneIdToLdMatrixMatrixCoord()
245 ArrayRef<int64_t> operandShape = params.fragmentType.getShape(); in getLaneIdToLdMatrixMatrixCoord()
253 if (params.contiguousDimType == IteratorType::Reduction) { in getLaneIdToLdMatrixMatrixCoord()
261 if (params.contiguousDimType == IteratorType::Parallel) { in getLaneIdToLdMatrixMatrixCoord()