Searched refs:WarpMatrixInfo (Results 1 – 3 of 3) sorted by relevance
31 struct WarpMatrixInfo { struct38 FailureOr<WarpMatrixInfo> getWarpMatrixInfo(Operation *op); argument43 int64_t inferTileWidthInBits(const WarpMatrixInfo &type);57 getMmaSyncRegisterType(const WarpMatrixInfo &type);68 const WarpMatrixInfo &fragmentType);78 FailureOr<LdMatrixParams> getLdMatrixParams(const WarpMatrixInfo &type,
34 int64_t inferNumRegistersPerMatrixFragment(const WarpMatrixInfo &type) { in inferNumRegistersPerMatrixFragment()54 FailureOr<WarpMatrixInfo> getWarpMatrixInfo(Operation *op) { in getWarpMatrixInfo()55 WarpMatrixInfo info; in getWarpMatrixInfo()87 int64_t inferTileWidthInBits(const WarpMatrixInfo &type) { in inferTileWidthInBits()100 getMmaSyncRegisterType(const WarpMatrixInfo &type) { in getMmaSyncRegisterType()174 const WarpMatrixInfo &fragmentType) { in getLaneIdAndValueIdToOperandCoord()208 FailureOr<nvgpu::LdMatrixParams> getLdMatrixParams(const WarpMatrixInfo &type, in getLdMatrixParams()
447 FailureOr<nvgpu::WarpMatrixInfo> warpMatrixInfo = in convertConstantOpMmaSync()473 FailureOr<nvgpu::WarpMatrixInfo> warpMatrixInfo = in creatLdMatrixCompatibleLoads()516 FailureOr<nvgpu::WarpMatrixInfo> warpMatrixInfo = in createNonLdMatrixLoads()605 FailureOr<nvgpu::WarpMatrixInfo> warpMatrixInfo = in convertTransferReadToLoads()638 FailureOr<nvgpu::WarpMatrixInfo> warpMatrixInfo = in convertTransferWriteToStores()